import asyncio
import websockets

async def handle_websocket(websocket, path):
# 当前端发出连接请求时，会进入这个函数处理
# 假设前端请求建立连接时会传递一些参数，这里可以进行处理
# ...
    await websocket.send("public_key")
    # 发送第二条消息
    await websocket.send("challenge")
    # 等待服务器关闭连接
    await websocket.wait_closed()

start_server = websockets.serve(handle_websocket, "localhost", 8900)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()